import torch

print(torch.__version__)          # 查看 PyTorch 版本
print(torch.cuda.is_available())  # 返回 True 表示 CUDA 可用
print(torch.version.cuda)         # 查看 PyTorch 使用的 CUDA 版本